import numpy as np

# 创建一个模拟的输入数据，形状为 (B, W, H, C)
# 假设 batch size = 2, 宽度 W = 4, 高度 H = 3, 通道数 C = 3
input_data = np.random.randint(0, 10, size=(2, 4, 3, 3))

# 打印输入数据的形状
print("input_data:", input_data.shape, "\ninput_data.shape: ", input_data.shape)

# 动态获取宽度、高度和通道数
# W, H, C = input_data.shape[1], input_data.shape[2], input_data.shape[3]
W, H, C = input_data.shape[1:]
# 打印宽度、高度和通道数
print("宽度 W:", W)
print("高度 H:", H)
print("通道数 C:", C)
